package cn.itcast.task;

import cn.itcast.bean.CleanBean;
import cn.itcast.function.KeyFunction;
import cn.itcast.function.StockKlineWindowFunction;
import cn.itcast.inter.ProcessDataInterface;
import cn.itcast.util.DorisUtil;
import org.apache.flink.streaming.api.datastream.DataStream;
import org.apache.flink.streaming.api.windowing.assigners.TumblingEventTimeWindows;
import org.apache.flink.streaming.api.windowing.time.Time;

/**
 * 个股K线(日,周,月)
 * 在此我们只实现日K
 */
public class StockDayKlineTask implements ProcessDataInterface {
    @Override
    public void process(DataStream<CleanBean> waterData) {

        /**
         * 开发步骤：
         * 1.数据分组
         * 2.划分时间窗口
         * 3.数据处理
         * 4.数据写doris
         */
        //1.数据分组
        waterData.keyBy(new KeyFunction())
                //2.划分时间窗口
                .window(TumblingEventTimeWindows.of(Time.seconds(5)))
                //3.数据处理
                .apply(new StockKlineWindowFunction())
                // 4.数据写doris
                .addSink(DorisUtil.getSinkFunction("insert into quot_stock_kline values(?,?,?,?,?,?,?,?,?,?,?,?,?)"));
    }
}
